from model.MyOllamaEmbeddings import MyOllamaEmbeddings

embeddings = MyOllamaEmbeddings(model="nomic-embed-text")

vectors = embeddings.embed_documents(
    [
        "嗨" ,
        "哦，你好",
        "你叫什么名字？",
        "我的朋友们的叫我World",
        "Hello word"
    ]
)

print(f"生成的向量数量: {len(vectors)}") # 文本数量
print(f"每个向量的维度: {len(vectors[0])}") # 第一段文本的embeddings向量维度

embeddings_query = embeddings.embed_query("对话中提到的名字是什么")
#查询嵌入的前5个值，就是相似度
print(embeddings_query[:5])